"""
118. 杨辉三角
已解答
简单
相关标签
premium lock icon
相关企业
给定一个非负整数 numRows，生成「杨辉三角」的前 numRows 行。

在「杨辉三角」中，每个数是它左上方和右上方的数的和。





示例 1:

输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
示例 2:

输入: numRows = 1
输出: [[1]]


提示:

1 <= numRows <= 30
"""
from typing import List


class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        ans = []
        # ans.append([1])
        for i in range(0,numRows):
            cur = []
            for j in range(i+1):
                if j == 0 or j == i:
                    cur.append(1)
                else:
                    cur.append(ans[i-1][j-1] + ans[i-1][j])
            ans.append(cur)
        return ans

if __name__ == '__main__':
    print(Solution().generate(5))
